from langchain_openai import ChatOpenAI
import random
import openai
from openai import OpenAI
import random

client = OpenAI(
    api_key="sk-rUdoFP3fG39PSAzFD35b0d2302864bAc9b8c4d40B3C16aF9",     
    base_url="http://llm.bnuzh.edu.cn:9080/v1",
)

def llm_solve_problem(input_context):
    chat_completion = client.chat.completions.create(
        messages=[
            {
                "role": "user",
                "content": input_context,
            }
        ],
        model="qwen25-72b",
    )
    response = chat_completion.choices[0].message.content
    print(response)
    return response

if __name__ == '__main__':
    llm_solve_problem('who are you')